package com.linzm.leetcode.primary.exercises3_20230122;

import java.util.Deque;
import java.util.LinkedList;

/**
 * @Author zimingl
 * @Date 2023/1/24 20:02
 * @Description: 重新排列单词间的空格
 */
public class Demo15_1592 {
    /**
     * 给你一个字符串 text ，该字符串由若干被空格包围的单词组成。每个单词由一个或者多个小写英文字母组成，
     * 并且两个单词之间至少存在一个空格。题目测试用例保证 text 至少包含一个单词 。
     * 请你重新排列空格，使每对相邻单词之间的空格数目都 相等 ，并尽可能 最大化 该数目。
     * 如果不能重新平均分配所有空格，请 将多余的空格放置在字符串末尾 ，这也意味着返回的字符串应当与原 text 字符串的长度相等。
     * 返回 重新排列空格后的字符串 。
     * 输入：text = "  this   is  a sentence "
     * 输出："this   is   a   sentence"
     * 解释：总共有 9 个空格和 4 个单词。可以将 9 个空格平均分配到相邻单词之间，相邻单词间空格数为：9 / (4-1) = 3 个。
     */
    public static void main(String[] args) {
        Demo15_1592 demo15_1592 = new Demo15_1592();
//        String text = "  this   is  a sentence ";
        String text = "  this ";
        String reorderSpaces = demo15_1592.reorderSpaces(text);
        System.out.println(reorderSpaces);
    }

    private String reorderSpaces(String text) {
        int blankCnt = 0, wordCnt = 0;
        for (int i = 0; i < text.length(); i++) {
            if (text.charAt(i) == ' ') {
                blankCnt++;
            }
        }
        Deque<String> queue = new LinkedList<>();
        String[] original = text.split(" ");
        for (int i = 0; i < original.length; i++) {
            if (!original[i].equals("")) {
                wordCnt++;
                queue.add(original[i]);
            }
        }
        String res = "";
        if (wordCnt == 1 && text.charAt(0) == ' ') {
            while (!queue.isEmpty()) {
                res += queue.remove();
                while (blankCnt > 0) {
                    res += " ";
                    blankCnt--;
                }
            }
            return  res;
        }
        if (wordCnt == 1) {
            return text;
        }
        int index = blankCnt / (wordCnt - 1);
        int end = blankCnt % (wordCnt - 1);
        while (!queue.isEmpty()) {
            res += queue.remove();
            while (index > 0 && queue.size() >= 1) {
                res += " ";
                index--;
            }
            index = blankCnt / (wordCnt - 1);
        }
        while (end > 0) {
            res += " ";
            end--;
        }
        return res;
    }
}
